package leetcode.L500;

import java.util.*;

/*
层序遍历
*/


public class EX102 {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();//输出的
        Queue<TreeNode> queue = new ArrayDeque<>();//临时队列
        if(root!=null){
            queue.add(root);
        }
        while (queue.size()>0){
            List<Integer> list = new ArrayList<>();//每一个小层
            int n = queue.size();//队列特性，后入前出
            for (int i = 0; i < n; i++) {
                TreeNode node =queue.poll();
                list.add(node.val);
                if(node.left!=null){
                    queue.add(node.left);
                }
                if(node.right!=null){
                    queue.add(node.right);
                }
            }
            res.add(list);
        }
        return res;
    }

    public List<List<Integer>> levelOrder2(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        Deque<TreeNode> queue = new LinkedList<>();
        queue.addFirst(root);
        while(queue!=null){
            List<Integer> temp = new ArrayList<>();
            int size = queue.size();
            for(int i = 0;i<size;i++){
                TreeNode node = queue.pollFirst();
                temp.add(node.val);
                if(node.left!=null){
                    queue.addLast(node.left);
                }
                if(node.right!=null){
                    queue.addLast(node.right);
                }
            }
            res.add(temp);
        }
        return res;

    }
}
